from enum import Enum


class StrategyType(Enum):
    VOLUME = ("VOLUME", "交易量")
    PRICE = (
        "PRICE",
        "交易价",
    )
    NEWS = ("NEWS", "消息面")
    TIME_POINT = ("TIME_POINT", "时间节点")
    COMPREHENSIVE = ("COMPREHENSIVE", "综合")

    def __init__(
        self,
        code: str,
        description: str,
    ):
        self.description = description  # 中文描述
        self.code = code  # Code 标识

    def __str__(self):
        return f"{self.code}: {self.description}"
